Generative Universal Multimodal Retrieval with Dual-role Identifiers
双角色标识符的生成式通用多模态检索
🧭 生成式多模态检索 · 双角色 Semantic ID ★ 深度解读
Independent Researcher, Japan · Institute of Library, Information and Media Science, University of Tsukuba, Japan · College of Knowledge and Library Sciences, University of Tsukuba, Japan
DrIG 将同一残差量化 ID 同时作为顺序解码序列和无序集合先验,缓解前缀剪枝错误,并以 dense 重排补偿量化损失
📌 工作参考价值(4 点)
- 双角色 Semantic ID 可直接用于生成式召回:将商品 ID 设计为分层量化 token,第一层编码类目/模态,其余层编码语义;推理时 beam search 不仅用 decoder 序列分数,还加入由 ID 无序集合计算的全局相关性分数,缓解 beam search 前缀剪枝误杀。
- 生成召回 + dense 精排是现实的混合架构:先用轻量生成式模型通过 Trie 约束 beam search 召回 top-k,再用 query/item embedding 余弦相似度重排,避免全库 ANN 打分,额外成本仅为 O(kd),适合大规模电商商品库。
- Query augmentation 与判别排序目标可迁移:用 query-target embedding 的 Beta 插值生成增强 query 稳定生成器训练;用 teacher 相似度计算 adaptive margin 的 pairwise ranking loss,使 decoder 序列分数更贴近排序目标,适合用于生成式商品召回的训练。
- LMM 作为多模态 item embedding 编码器:用 “summarize in one word” 提示提取 <emb> 前 hidden state 作为通用多模态表示,再做多任务对比微调,可复用到图文商品/内容理解与跨模态召回。